package com.gxc.tree;

/**
 * 给你一棵二叉树的根节点，返回该树的 直径 。
 *
 * 二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root 。
 *
 * 两节点之间路径的 长度 由它们之间边数表示。
 */
public class DiameterOfBinaryTree {

      public class TreeNode {
      int val;
      TreeNode left;
      TreeNode right;
      TreeNode() {}
      TreeNode(int val) { this.val = val; }
      TreeNode(int val, TreeNode left, TreeNode right) {
          this.val = val;
          this.left = left;
          this.right = right;
      }
  }

    public int diameterOfBinaryTree(TreeNode root) {
        int[] result = treeHigh(root);
        return result[2];
    }

    private int[] treeHigh(TreeNode root) {
          if (root == null) return new int[]{0,0,0};

          int[] left = treeHigh(root.left);
          int leftMax = Math.max(left[0], left[1]);
          int[] right = treeHigh(root.right);
          int rightMax = Math.max(right[0], right[1]);
          int max = Math.max(left[2], right[2]);
          return new int[]{leftMax + 1, rightMax + 1, Math.max(leftMax + rightMax, max)};
    }

    int ans;
    public int diameterOfBinaryTree2(TreeNode root) {
        ans = 1;
        depth(root);
        return ans - 1;
    }
    public int depth(TreeNode node) {
        if (node == null) {
            return 0; // 访问到空节点了，返回0
        }
        int L = depth(node.left); // 左儿子为根的子树的深度
        int R = depth(node.right); // 右儿子为根的子树的深度
        ans = Math.max(ans, L+R+1); // 计算d_node即L+R+1 并更新ans
        return Math.max(L, R) + 1; // 返回该节点为根的子树的深度
    }

}
